package com.tt.common.page;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.io.Serializable;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 通用返回数据
 * @author 青青
 */
@Data
public class TtResult implements Serializable
{
    // 定义jackson对象
    private static final ObjectMapper MAPPER = new ObjectMapper();

    /**
     * 响应业务状态
     */
    private Integer code;

    /**
     * 响应消息
     */
    private String msg;

    /**
     * 响应中的数据
     */
    private Object data;

    /**
     * 总记录数
     */
    private int total;

    /**
     * 静态方法
     * @param status
     * @param msg
     * @param data
     * @return
     */
    public static TtResult build(Integer status, String msg, Object data)
    {
        return new TtResult(status, msg, data);
    }

    public static TtResult ok(Object data)
    {
        return new TtResult(data);
    }

    public static TtResult ok()
    {
        return new TtResult(null);
    }

    /**
     *
     */
    public TtResult() {}

    public static TtResult build(Integer status, String msg)
    {
        return new TtResult(status, msg, null);
    }

    /**
     * 添加额外数据返回
     * @param key
     * @param value
     * @return
     */
    public static TtResult push(String key, Object value)
    {
        Map<String, Object> map = new HashMap<>();
        map.put(key,value);
        return new TtResult(map);
    }

    /**
     * 构造方法
     * @param status
     * @param msg
     * @param data
     */
    public TtResult(Integer status, String msg, Object data)
    {
        this.code = status;
        this.msg = msg;
        this.data = data;
    }

    /**
     * 构造方法
     * @param data
     */
    public TtResult(Object data)
    {
        this.code = 0;
        this.msg = "OK";
        this.data = data;
    }



    /**
     * 将json结果集转化为TtResult对象
     * @param jsonData json数据
     * @param clazz TtResult中的object类型
     * @return
     */
    public static TtResult formatToPojo(String jsonData, Class<?> clazz) {
        try {
            if (clazz == null) {
                return MAPPER.readValue(jsonData, TtResult.class);
            }
            JsonNode jsonNode = MAPPER.readTree(jsonData);
            JsonNode data = jsonNode.get("data");
            Object obj = null;
            if (clazz != null) {
                if (data.isObject()) {
                    obj = MAPPER.readValue(data.traverse(), clazz);
                } else if (data.isTextual()) {
                    obj = MAPPER.readValue(data.asText(), clazz);
                }
            }
            return build(jsonNode.get("status").intValue(), jsonNode.get("msg").asText(), obj);
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * 没有object对象的转化
     *
     * @param json
     * @return
     */
    public static TtResult format(String json)
    {
        try
        {
            return MAPPER.readValue(json, TtResult.class);
        } catch (Exception e)
        {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * Object是集合转化
     *
     * @param jsonData json数据
     * @param clazz 集合中的类型
     * @return
     */
    public static TtResult formatToList(String jsonData, Class<?> clazz)
    {
        try
        {
            JsonNode jsonNode = MAPPER.readTree(jsonData);
            JsonNode data = jsonNode.get("data");
            Object obj = null;
            if (data.isArray() && data.size() > 0)
            {
                obj = MAPPER.readValue(data.traverse(),
                        MAPPER.getTypeFactory().constructCollectionType(List.class, clazz));
            }
            return build(jsonNode.get("status").intValue(), jsonNode.get("msg").asText(), obj);
        } catch (Exception e)
        {
            return null;
        }
    }

}
